#!/bin/bash

# 容器环境下的数据库初始化脚本
# 这个脚本是专门为Docker容器环境设计的简化版本

# 设置颜色变量
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# 输出信息函数
echo_info() {
    echo -e "${BLUE}[信息]${NC} $1"
}

echo_success() {
    echo -e "${GREEN}[成功]${NC} $1"
}

echo_warning() {
    echo -e "${YELLOW}[警告]${NC} $1"
}

echo_error() {
    echo -e "${RED}[错误]${NC} $1"
}

# 设置工作目录
cd /app/scripts

# 确保go.mod文件存在
echo_info "检查Go模块配置..."
go mod init github.com/xianxiagame/scripts || true

echo_info "执行go mod tidy安装依赖..."
export GOPROXY="https://goproxy.cn,direct"
go mod tidy
if [ $? -ne 0 ]; then
    echo_error "依赖安装失败，请检查网络连接"
    exit 1
fi

echo_success "所有依赖包安装成功"

# 运行数据库初始化脚本
echo_info "运行数据库初始化脚本..."
go run init_database.go
if [ $? -ne 0 ]; then
    echo_error "数据库初始化失败，请查看错误日志"
    exit 1
fi

echo_success "数据库初始化完成！"